1043C - Smallest Word - CodeForces Solution

#include <bits/stdc++.h>

#define _PRINT_ 1
#define _TEST_ 1
#define _SOLVE_ 1

#if defined(DEBUG) && _PRINT_ == 1
#include "debug.h"
#else
#define print(x, ...) 0
#define debug(x, ...) 0
#define verify(x, ...) 0
#endif

using namespace std;

using ll = long long;
constexpr int inf = 1e9;
constexpr ll llinf = 1e18;

/**
 *
 */
void solve() {
    string s;
    cin >> s;
    int n = s.size();
    vector<int> ans(n);
    for (int i = 0; i < n; i++) {
        if (s[i] == 'a') {
            if (i == n - 1 || s[i + 1] == 'b') {
                ans[i] = 1;
            }
        } else {
            if (i < n - 1 && s[i + 1] == 'a') {
                ans[i] = 1;
            }
        }
    }
    for (auto x : ans) {
        cout << x << " ";
    }
}

int main() {
#ifdef DEBUG
    freopen("input.txt", "r", stdin);
#endif
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t = 1;
//    cin >> t;
    while(t--) {
        solve();
    }
    return 0;
}